/**
 * Software License, Version 1.0
 * 
 * Copyright 2003 The Trustees of Indiana University. All rights reserved.
 * 
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 * 
 * 1) All redistributions of source code must retain the above copyright notice,
 * the list of authors in the original source code, this list of conditions and
 * the disclaimer listed in this license; 2) All redistributions in binary form
 * must reproduce the above copyright notice, this list of conditions and the
 * disclaimer listed in this license in the documentation and/or other materials
 * provided with the distribution; 3) Any documentation included with all
 * redistributions must include the following acknowledgement:
 * 
 * "This product includes software developed by the Community Grids Lab. For
 * further information contact the Community Grids Lab at
 * http://communitygrids.iu.edu/."
 * 
 * Alternatively, this acknowledgement may appear in the software itself, and
 * wherever such third-party acknowledgments normally appear.
 * 
 * 4) The name Indiana University or Community Grids Lab or NaradaBrokering,
 * shall not be used to endorse or promote products derived from this software
 * without prior written permission from Indiana University. For written
 * permission, please contact the Advanced Research and Technology Institute
 * ("ARTI") at 351 West 10th Street, Indianapolis, Indiana 46202. 5) Products
 * derived from this software may not be called NaradaBrokering, nor may Indiana
 * University or Community Grids Lab or NaradaBrokering appear in their name,
 * without prior written permission of ARTI.
 * 
 * 
 * Indiana University provides no reassurances that the source code provided
 * does not infringe the patent or any other intellectual property rights of any
 * other entity. Indiana University disclaims any liability to any recipient for
 * claims brought by any other entity based on infringement of intellectual
 * property rights or otherwise.
 * 
 * LICENSEE UNDERSTANDS THAT SOFTWARE IS PROVIDED "AS IS" FOR WHICH NO
 * WARRANTIES AS TO CAPABILITIES OR ACCURACY ARE MADE. INDIANA UNIVERSITY GIVES
 * NO WARRANTIES AND MAKES NO REPRESENTATION THAT SOFTWARE IS FREE OF
 * INFRINGEMENT OF THIRD PARTY PATENT, COPYRIGHT, OR OTHER PROPRIETARY RIGHTS.
 * INDIANA UNIVERSITY MAKES NO WARRANTIES THAT SOFTWARE IS FREE FROM "BUGS",
 * "VIRUSES", "TROJAN HORSES", "TRAP DOORS", "WORMS", OR OTHER HARMFUL CODE.
 * LICENSEE ASSUMES THE ENTIRE RISK AS TO THE PERFORMANCE OF SOFTWARE AND/OR
 * ASSOCIATED MATERIALS, AND TO THE PERFORMANCE AND VALIDITY OF INFORMATION
 * GENERATED USING SOFTWARE.
 */
package cgl.narada.event.impl;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;

import cgl.narada.event.DistributionTraces;

/**
 * Provides information regarding the distribution traces associated with an
 * event.
 * 
 * @author Shrideep Pallickara $Date$ $Revision$
 */

public class DistributionTracesImpl implements DistributionTraces {
   private byte[] traversedSoFar = null;

   private byte[] toReach = null;

   private String moduleName = "DistributionTracesImpl:";

   public DistributionTracesImpl() {
      traversedSoFar = new byte[16];
      toReach = new byte[16];
   }

   public DistributionTracesImpl(byte[] tracesBytes) {
      if (tracesBytes.length == 1) {
         return;
      }
      ByteArrayInputStream baInputStream = new ByteArrayInputStream(tracesBytes);
      DataInputStream din = new DataInputStream(baInputStream);
      try {
         int offset = (int) din.readByte();
         boolean _toReach = true;
         boolean _traversed = true;
         if (offset == 1) {
            _traversed = false;
         }

         if (offset == 2) {
            _toReach = false;
         }

         if (_toReach) {
            int _reachLength = din.readShort();
            toReach = new byte[_reachLength];
            din.readFully(toReach);
         }

         if (_traversed) {
            int _traversedLength = din.readShort();
            traversedSoFar = new byte[_traversedLength];
            din.readFully(traversedSoFar);
         }

      } catch (IOException ioe) {
         System.out
                   .println(moduleName + "Error while un-marshalling");
      }
      /** end try-catch block */
   }

   public void setDestinationsTraversedSoFar(byte[] traversedSoFar) {
      this.traversedSoFar = traversedSoFar;
   }

   /** Get the destinations traversed so far */
   public byte[] getDestinationsTraversedSoFar() {
      return traversedSoFar;
   }

   /** Setting the destinations to reach */
   public void setDestinationsToReach(byte[] toReach) {
      this.toReach = toReach;
   }

   /** Get the destinations to reach */
   public byte[] getDestinationsToReach() {
      return toReach;
   }

   public byte[] getBytes() {
      ByteArrayOutputStream baOutputStream = new ByteArrayOutputStream();
      DataOutputStream dout = new DataOutputStream(baOutputStream);
      byte[] marshalledBytes = null;

      try {
         int offset = 0;

         if (toReach != null) {
            offset += 1;
         }

         if (traversedSoFar != null) {
            offset += 2;
         }

         dout.writeByte(offset);
         if (offset == 0) {
            marshalledBytes = baOutputStream.toByteArray();
            return marshalledBytes;
         }

         if (toReach != null) {
            dout.writeShort((short) toReach.length);
            dout.write(toReach);
         }

         if (traversedSoFar != null) {
            dout.writeShort((short) traversedSoFar.length);
            dout.write(traversedSoFar);
         }

         dout.flush();
         marshalledBytes = baOutputStream.toByteArray();
      } catch (IOException ioe) {
         System.out
                   .println(moduleName + "Error while marshalling");
      }
      /** end try-catch block */
      return marshalledBytes;
   }

   public static void main(String[] args) {
      byte[] toReach = new String("123456789").getBytes();
      byte[] traversedSoFar = new String("987654321").getBytes();

      // traversedSoFar = null;
      // toReach = null;

      DistributionTracesImpl traces = new DistributionTracesImpl();

      traces.setDestinationsToReach(toReach);
      traces.setDestinationsTraversedSoFar(traversedSoFar);

      byte[] marshalledBytes = traces.getBytes();

      DistributionTracesImpl traces2 = new DistributionTracesImpl(
                                                                  marshalledBytes);

      toReach = traces2.getDestinationsToReach();
      traversedSoFar = traces2.getDestinationsTraversedSoFar();

      if (toReach != null)
         System.out.println("ToReach " + new String(toReach));
      if (traversedSoFar != null)
         System.out.println("TraversedSoFar " + new String(traversedSoFar));

   }

}
